# coding: utf-8

import matplotlib.pyplot as plt
import numpy as np
import matplotlib

font = {"family": "Arial", "weight": "normal", "size": 14}
matplotlib.rc("font", **font)
LineWidth = 3

# ECCV 2024: WeConvene: Learned Image Compression with  Wavelet-Domain Convolution and Entropy Model
bpp = [0.163, 0.203, 0.309, 0.452, 0.637, 0.897]
psnr = [30.07, 30.88, 32.53, 34.29, 36.03, 37.96]
weconvene, = plt.plot(bpp, psnr, "k-o", LineWidth=LineWidth, label="WeConvene")



# NOTE: 最终的显示
plt.legend(handles=[weconvene], loc=4)
plt.grid()
plt.xlabel('Bpp')
plt.ylabel('PSNR(dB)')
plt.title('Kodak dataset')
plt.savefig('kodak_psnr.png', format='png', dpi=600, bbox_inches='tight')
